package com.bestcem.xm.txtanls.dao.mapper;

import com.bestcem.xm.txtanls.entity.mysql.TrainModelDo;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;

import java.util.List;

@Mapper
public interface TrainModelMapper {

    /**
     * 获取训练模型列表
     *
     * @return
     */
    List<TrainModelDo> listTrainModel();

    /**
     * 根据code查询
     *
     * @param code 训练模型code，与行业code一致
     * @return
     */
    TrainModelDo getByCode(@Param("code") Integer code);


    /**
     * 根据id查询
     *
     * @param trainModelId 训练模型id
     * @return
     */
    TrainModelDo getById(@Param("trainModelId") String trainModelId);
}
